#!/usr/bin/env python3
"""
Clean up old output files and prepare for a fresh simulation run.
"""

import os
import shutil

def clean_output():
    """Remove old output files and dummy mesh."""
    
    # Directories and files to clean
    items_to_clean = [
        "output/",
        "output/visualization/",
    ]
    
    for item in items_to_clean:
        if os.path.exists(item):
            if os.path.isdir(item):
                print(f"Removing directory: {item}")
                shutil.rmtree(item)
            else:
                print(f"Removing file: {item}")
                os.remove(item)
    
    print("\nCleanup complete!")
    print("You can now run:")
    print("  1. python generate_mesh.py  (to create proper mesh)")
    print("  2. python test_fixes.py     (to verify everything works)")
    print("  3. python main.py           (to run simulation)")

if __name__ == "__main__":
    clean_output() 